# -*- coding: utf-8 -*-
# Author: xiantingDeng
# File: handler.py
# Time: 16:55

from common.element.elements import PocoElement
from common.util.api_util import get_yaml_data


class TransactionHandler(PocoElement):
    def __init__(self, poco, log):
        super().__init__(poco, log)
        self.page = get_yaml_data("login.yaml")

    def get_shuakashoukuan(self):
        return self.poco_element_exists(self.page.get("shuakashoukuan"))


    def get_shoukuanjine(self):
        return self.poco_element_exists(self.page.get("shoukuanjine"))


    def get_querenshoukuan(self):
        return self.poco_element_exists(self.page.get("querenshoukuan"))